# -*- coding:utf-8 -*-

from base.main import train, predict
from utils.io_utils import convert_abspath

"""
LeNet5 模型测试
"""

drnn_train_config = convert_abspath('config/core/train/drnn.yml')
drnn_predict_config = convert_abspath('config/core/predict/drnn.yml')


def drnn_train():
    """
    LeNet5 训练
    :return:
    """
    train(drnn_train_config)


def drnn_predict():
    """
    LeNet5 预测
    :return:

    """
    # 抽取两条
    x = '天'
    # 预测
    y_pred = predict(drnn_predict_config, x)
    print(y_pred)


if __name__ == '__main__':
    # drnn_train()
    drnn_predict()
